from llama_index.core.vector_stores.types import BasePydanticVectorStore
from llama_index.vector_stores.azurecosmosmongo import AzureCosmosDBMongoDBVectorSearch


def test_class():
    names_of_base_classes = [
        b.__name__ for b in AzureCosmosDBMongoDBVectorSearch.__mro__
    ]
    assert BasePydanticVectorStore.__name__ in names_of_base_classes
